package com.example;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Semaphore;

public class ConcurrentTest {

    private static int thread_num = 200;

    private static int client_num = 460;

    private static Map keywordMap = new HashMap();

    static {
        try {
            InputStreamReader isr = new InputStreamReader(new FileInputStream(new File("D:/test.txt")), "UTF-8");
            BufferedReader buffer = new BufferedReader(isr);
            String line = "";
            while ((line = buffer.readLine()) != null) {
                keywordMap.put(line.substring(0, line.lastIndexOf(":")), "");
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        int size = keywordMap.size();

        ExecutorService exec = Executors.newCachedThreadPool();

        // 50个线程可以同时访问
        final Semaphore semp = new Semaphore(thread_num);

        // 模拟2000个客户端访问
        for (int index = 0; index < client_num; index++) {
            final int NO = index;
            Runnable run = new Runnable() {
                @Override
                public void run() {
                    try {
                        // 获取许可
                        semp.acquire();
                        System.out.println("Thread:" + NO);
//                        String host = "http://10.99.23.42:7001/KMQueryCenter/query.do?";
//                        String para = "method=getQueryResult&pageNum=1&pageSize=5&"
//                                + "queryKeyWord="
//                                + getRandomSearchKey(NO)
//                                + "&questionID=-1&questionIdPath=-1&searchType=1"
//                                + "&proLine=&proSeries=&proType=" + NO;
                        String host = "http://www.baidu.com";
                        String para = "?";
                        System.out.println(host + para);

                        URL url = new URL(host);// 此处填写供测试的url

                        HttpURLConnection connection = (HttpURLConnection) url.openConnection();

//                        connection.setRequestMethod("POST");
//                        connection.setRequestProperty("Proxy-Connection", "Keep-Alive");

                        connection.setDoOutput(true);
                        connection.setDoInput(true);

                        PrintWriter out = new PrintWriter(connection.getOutputStream());
                        out.print(para);
                        out.flush();
                        out.close();

                        BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
                        String line = "";
                        String result = "";
                        while ((line = in.readLine()) != null) {
                            result += line;
                        }

                        // System.out.println(result);
                        // Thread.sleep((long) (Math.random()) * 1000);
                        // 释放
                        System.out.println("第：" + NO + " 个");
                        semp.release();
                    } catch (Exception e) {
                        e.printStackTrace();
                    }
                }
            };
            exec.execute(run);
        }

        // 退出线程池
        exec.shutdown();
    }

    private static String getRandomSearchKey(final int no) {
        String ret = "";
        int size = keywordMap.size();
        // int wanna = (int) (Math.random()) * (size - 1);
        ret = (keywordMap.entrySet().toArray())[no].toString();
        ret = ret.substring(0, ret.lastIndexOf("="));
        System.out.println("\t" + ret);
        return ret;
    }

}
